tensor([[[0.3500, 0.3500],
         [0.3500, 0.3500]],

        [[0.3100, 0.3100],
         [0.3100, 0.3100]],

        [[0.3500, 0.3500],
         [0.3500, 0.3500]]], grad_fn=<RoundBackward1>)